click
matplotlib
numpy
torch
torchvision
tqdm
pyyaml
wandb
einops
transformers[sentencepiece]
sentencepiece
rotary-embedding-torch
streamlit